import pandas as pd
import os
import numpy as np

if __name__ == '__main__':
    base_dir = './intermediate-rating-csv'
    dataset_l = ['yahoomusic_big']
    for ds in dataset_l:
        csv = pd.read_csv(os.path.join(base_dir, '%s.csv' % ds))
        n_user = np.max(csv['userID'])
        n_item = np.max(csv['itemID'])
        print("ds {}, n_user {}, n_item {}".format(ds, n_user, n_item))
